import os
import zlib
import zipfile


def compression(data: bytes) -> bytes:
    """
    数据压缩处理
    :param data: 二进制数据
    :return: 二进制数据
    """
    zlib_encode = zlib.compressobj(zlib.Z_BEST_COMPRESSION)
    data_bytes = zlib_encode.compress(data)
    data_bytes += zlib_encode.flush()
    return data_bytes


def decompression(data: bytes) -> bytes:
    """
    数据解压缩处理
    :param data: 二进制数据
    :return: 二进制数据
    """
    zlib_decode = zlib.decompressobj()
    data_bytes = zlib_decode.decompress(data)
    data_bytes += zlib_decode.flush()
    return data_bytes
